package com.zaey.auto.utils;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.FilenameUtils;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.List;

/**
 * @author JohnSoybean
 * @create 2020-05-23 3:16
 * @description
 */
public class WaterUtils {

    /**
     * 给图片添加水印文字、可设置水印文字的旋转角度
     *
     * @param srcImgPath
     * @param targerPath
     * @param degree
     */
    public static void markImageByText(String text1, String text2, String srcImgPath, String targerPath, int x, int y, float alpha,
                                       Integer degree) {

        InputStream is = null;
        OutputStream os = null;
        try {
            // 1、源图片
            Image srcImg = ImageIO.read(new File(srcImgPath));
            BufferedImage buffImg = new BufferedImage(srcImg.getWidth(null), srcImg.getHeight(null),
                    BufferedImage.TYPE_INT_RGB);

            // 2、得到画笔对象
            Graphics2D g = buffImg.createGraphics();
            // 3、设置对线段的锯齿状边缘处理
            g.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BILINEAR);
            g.drawImage(srcImg.getScaledInstance(srcImg.getWidth(null), srcImg.getHeight(null), Image.SCALE_SMOOTH), 0,
                    0, null);
            // 4、设置水印旋转
            if (null != degree) {
                g.rotate(Math.toRadians(degree), (double) buffImg.getWidth() / 2, (double) buffImg.getHeight() / 2);
            }
            // 5、设置水印文字颜色
            g.setColor(Color.red);
            // 6、设置水印文字Font
            Font font = new Font(Font.SANS_SERIF, Font.BOLD, 50);
            g.setFont(font);
            // 7、设置水印文字透明度
            g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha));
            // 8、第一参数->设置的内容，后面两个参数->文字在图片上的坐标位置(x,y)
            g.drawString(text1, x, y);
            g.drawString(text2, x, y+100);
            // 9、释放资源
            g.dispose();
            // 10、生成图片
            os = new FileOutputStream(targerPath);
            ImageIO.write(buffImg, "JPG", os);

            System.out.println("图片完成添加水印文字");

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if (null != is)
                    is.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
            try {
                if (null != os)
                    os.close();
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    public static void addWaterRecordToday(String target, String last) {
        java.util.List<File> fileList = (List<File>) FileUtils.listFiles(new File(target), new String[]{"jpg","jpeg","png","bmp"}, false);
        for (File file : fileList) {
            markImageByText("0579", "", file.getAbsolutePath(), last + "/" + file.getName(), 600, 1100, 1, null);
        }
    }

    public static void addWaterRecordDetail(String target, String last) {
        java.util.List<File> fileList = (List<File>) FileUtils.listFiles(new File(target), new String[]{"jpg","jpeg","png","bmp"}, false);
        for (File file : fileList) {
            String fileName = file.getName();
            String num = null;
            System.out.println(fileName);
            if (fileName.contains("-")) {
                String[] split = fileName.split("-");
                num = split[0];
            } else {
                String[] split = fileName.split("\\.");
                num = split[0];
            }
            markImageByText("0579", num + "个赞", file.getAbsolutePath(), last + "/" + file.getName(), 600, 1100, 1, null);

        }
    }


    public static void main(String[] args) {
//        addWaterRecordToday("G:\\var\\tgt\\微信主页", "G:\\var\\last\\微信主页");
        addWaterRecordDetail("G:\\var\\tgt\\霸权时代", "G:\\var\\last\\霸权时代");
//        addWaterRecordDetail("G:\\var\\tgt\\不要低估", "G:\\var\\last\\不要低估");
    }

}
